package com.dang.alex.servlet;

import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import com.sun.image.codec.jpeg.JPEGCodec;
import com.sun.image.codec.jpeg.JPEGImageEncoder;

import java.awt.image.BufferedImage;
import java.io.IOException;
import java.util.Map;

import javax.servlet.*;

@WebServlet("/checkCode.action")
public class SafeCodeServlet extends HttpServlet{private static final long serialVersionUID = -2971417344472546855L;
@Override
protected   void   doGet(HttpServletRequest   req,   HttpServletResponse   resp)   throws ServletException, IOException {

	doPost(req, resp);
	}
@Override
protected   void   doPost(HttpServletRequest   req,   HttpServletResponse   resp)   throws ServletException, IOException {
	HttpSession session = req.getSession();Map<String,BufferedImage> map = com.dang.alex.util.ImageUtil.createImage();
	String key = map.keySet().iterator().next();session.setAttribute("checkCode", key);
	BufferedImage image = map.get(key);
	ServletOutputStream out = resp.getOutputStream();
	JPEGImageEncoder imageEncoder = JPEGCodec.createJPEGEncoder(out);
	imageEncoder.encode(image);out.flush(); 
	}
}